package com.dfsoft.util;

import lombok.extern.slf4j.Slf4j;

import java.io.*;
import java.util.Objects;

@Slf4j
public class FileUtils {
    /**
     * 将InputStream写入本地文件
     * @param destination 写入本地目录
     * @param input 输入流
     * @throws IOException IOException
     */
    public static void inputStreamWriteToLocal(String destination, InputStream input){
        createDir(destination);
        int index;
        byte[] bytes = new byte[1024];
        try (FileOutputStream downloadFile = new FileOutputStream(destination)){
            while ((index = input.read(bytes)) != -1) {
                downloadFile.write(bytes, 0, index);
                downloadFile.flush();
            }
//            input.close();
        }catch (IOException e){
            e.printStackTrace();
        }
    }

    /**
     * 将ByteArrayOutputStream写入本地文件
     * @param destination 写入本地目录
     * @param bos 输出流
     * @throws IOException IOException
     */
    public static void byteArrayOutputStreamWriteToLocal(String destination, ByteArrayOutputStream bos){
        createDir(destination);
        try (FileOutputStream downloadFile = new FileOutputStream(destination)){
            downloadFile.write(bos.toByteArray());
        }catch (IOException e){
            e.printStackTrace();
        }
    }

    public static void createDir(String destination){
        File file = new File(destination);
        if (!file.exists()) {
            file.getParentFile().mkdirs();
        }
    }

    /**
     * 将InputStream 写入 ByteArrayOutputStream
     * @param bos 写入
     * @param inputStream 输入流
     */
    public static void inputStreamWriteOutputStream(ByteArrayOutputStream bos, InputStream inputStream)throws IOException{
        byte[] buffer = new byte[1024];
        int length;
        while ((length = inputStream.read(buffer)) > 0) {
            bos.write(buffer, 0, length);
        }
    }

    /**
     * 文件是否存在
     * @param realPath
     * @return
     */
    public static boolean existsFile(String realPath){
        File file = new File(realPath);
        return file.exists();
    }

    /**
     * 删除目录及目录下的文件
     *
     * @param dir 要删除的目录的文件路径
     * @return 目录删除成功返回true，否则返回false
     */
    public static boolean deleteDirectory(String dir) {
        // 如果dir不以文件分隔符结尾，自动添加文件分隔符
        if (!dir.endsWith(File.separator)) {
            dir = dir + File.separator;
        }
        File dirFile = new File(dir);
        // 如果dir对应的文件不存在，或者不是一个目录，则退出
        if ((!dirFile.exists()) || (!dirFile.isDirectory())) {
            log.info("删除目录失败：" + dir + "不存在！");
            return false;
        }
        boolean flag = true;
        // 删除文件夹中的所有文件包括子目录
        File[] files = dirFile.listFiles();
        for (int i = 0; i < Objects.requireNonNull(files).length; i++) {
            // 删除子文件
            if (files[i].isFile()) {
                flag = deleteFileByName(files[i].getAbsolutePath());
                if (!flag) {
                    break;
                }
            } else if (files[i].isDirectory()) {
                // 删除子目录
                flag = deleteDirectory(files[i].getAbsolutePath());
                if (!flag) {
                    break;
                }
            }
        }

        if (!dirFile.delete() || !flag) {
            log.info("删除目录失败！");
            return false;
        }
        return true;
    }

    /**
     * 删除单个文件
     *
     * @param fileName 要删除的文件的文件名
     * @return 单个文件删除成功返回true，否则返回false
     */
    public static boolean deleteFileByName(String fileName) {
        File file = new File(fileName);
        // 如果文件路径所对应的文件存在，并且是一个文件，则直接删除
        if (file.exists() && file.isFile()) {
            if (file.delete()) {
                log.info("删除单个文件" + fileName + "成功！");
                return true;
            } else {
                log.info("删除单个文件" + fileName + "失败！");
                return false;
            }
        } else {
            log.info("删除单个文件失败：" + fileName + "不存在！");
            return false;
        }
    }
}
